CUDA_VISIBLE_DEVICES=0 python link_predict.py --device 0 --use_norm True --dataset LastFM_magnn